Skip to content

mlir: add enzyme.atomic_rmw op with ordering attribute#2950

Merged
wsmoses merged 1 commit into
mainfrom
jmp/enzyme-atomic-rmw
Jul 22, 2026
Merged

mlir: add enzyme.atomic_rmw op with ordering attribute#2950
wsmoses merged 1 commit into
mainfrom
jmp/enzyme-atomic-rmw

Conversation

@pengmai

@pengmai pengmai commented Jul 15, 2026

Copy link
Copy Markdown
Member

No description provided.

@pengmai
pengmai requested a review from wsmoses July 15, 2026 20:51
Comment thread enzyme/Enzyme/MLIR/Dialect/EnzymeOps.td
enzyme::AtomicRMWOp::create(
builder, loadOp.getLoc(), gradient.getType(),
arith::AtomicRMWKind::addf, Ordering::monotonic, gradient,
memrefGradient, indices);

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we also need to do the same for the enzyme::AffineAtomicRMWOp too

@pengmai
pengmai force-pushed the jmp/enzyme-atomic-rmw branch from d921722 to 8e3f5c5 Compare July 22, 2026 18:26
@wsmoses
wsmoses merged commit f19caab into main Jul 22, 2026
37 checks passed
@wsmoses
wsmoses deleted the jmp/enzyme-atomic-rmw branch July 22, 2026 18:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants